package cn.kgc.filter;

import com.netflix.zuul.ZuulFilter;
import com.netflix.zuul.context.RequestContext;
import org.apache.commons.lang.StringUtils;
import org.springframework.cloud.netflix.zuul.filters.support.FilterConstants;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpServletRequest;

@Component
public class TokenFilter extends ZuulFilter {

    /**
     * 过滤的类型
     * 返回pre为前置过滤，表示会在请求被路由之前执行
     * @return
     */
    @Override
    public String filterType() {
        return FilterConstants.PRE_TYPE;
    }

    /**
     * 过滤器执行顺序
     * 数字越小，优先级越高
     * @return
     */
    @Override
    public int filterOrder() {
        return 0;
    }

    /**
     * 该过滤器是否需要被执行
     * true-执行
     * false-不执行
     * url 包含/api/就会被登录拦截校验
     * @return
     */
    @Override
    public boolean shouldFilter() {
        RequestContext ctx = RequestContext.getCurrentContext();
        HttpServletRequest request = ctx.getRequest();

        //获取请求路径
        String url = request.getRequestURL().toString();
        System.out.println("用户请求地址："+url);

        //路径中包含"/view/",则执行该过滤器
        return url.contains("/api/");
    }

    /**
     * 过滤器的具体处理
     * 内部调用这个方法
     * @return
     */
    @Override
    public Object run() {
        System.out.println("ZullFilter----服务网关过滤器开始处理----------");

        RequestContext ctx = RequestContext.getCurrentContext();
        HttpServletRequest request = ctx.getRequest();
        String accessToken = request.getHeader("token");
        if (StringUtils.isEmpty(accessToken)) {
            ctx.setSendZuulResponse(false);//过滤该请求，不对其进行路由
            ctx.setResponseStatusCode(401);//响应状态码
            ctx.getResponse().setContentType("text/html;charset=utf-8");//响应内容编码
            ctx.setResponseBody("401! 访问拒绝，请先登录");//响应内容
        }
        return null;
    }
}